package com.atguigu.mq.config;

import jakarta.annotation.PostConstruct;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.ReturnedMessage;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

@Component
@Slf4j
public class MQProducerAckConfig implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnsCallback {

    @Autowired
    RabbitTemplate rabbitTemplate; //这个模板对象是SpringBoot自动化配置生效，自动创建。  给模板对象注册两个函数。这样发送消息后可以执行回调处理。

    /**
     * @PostConstruct注解是Java中的一个标准注解，
     * 它用于指定在对象创建之后立即执行的方法。
     * 当使用依赖注入（如Spring框架）或者其他方式创建对象时，@PostConstruct注解可以确保在对象完全初始化之后，执行相应的方法。
     *
     * 使用@PostConstruct注解的方法必须满足以下条件：
     *
     * 1. 方法不能有任何参数
     * 2. 方法必须是非静态的
     * 3. 方法不能返回任何值
     *
     * 当容器实例化一个带有@PostConstruct注解的Bean时，它会在调用构造函数之后，并在依赖注入完成之前调用被@PostConstruct注解标记的方法。
     * 这样，我们可以在该方法中进行一些初始化操作，比如读取配置文件、建立数据库连接等。
     */
    @PostConstruct // IOC容器初始化完成会调用此方法，对模板对象进一步初始化。
    public void init(){
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnsCallback(this);
    }

    /**
     * 确认回调函数
     * @param correlationData 关联数据
     * @param ack 交换机是否收到消息
     * @param cause 交换机没收到消息原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {
        log.info("correlationData="+correlationData); //关联数据，如果发生消息时携带了关联数据，会被返回。如果发消息时没携带关联对象，则返回null
        if(ack){
            log.info("交换机【收到】消息了");
        }else{
            log.info("交换机【没收到】消息了");
            log.error("cause="+cause);  //交换机没收到消息的原因。
        }
    }

    /**
     * 回退函数：  只有队列没收到消息才会执行回退函数。如果队列收到消息了，则不执行回退函数。
     * @param returned 退回的消息信息(包含数据本身和头信息等)
     */
    @Override
    public void returnedMessage(ReturnedMessage returned) {
        log.info("消息主体: " + new String(returned.getMessage().getBody()));
        log.info("应答码: " + returned.getReplyCode());
        log.info("描述：" + returned.getReplyText());
        log.info("消息使用的交换器 exchange : " + returned.getExchange());
        log.info("消息使用的路由键 routing : " + returned.getRoutingKey());
    }
}
